package com.yb.xczx.cloud.media.handler.xxljob;

import com.xxl.job.core.context.XxlJobHelper;
import com.xxl.job.core.handler.annotation.XxlJob;
import lombok.extern.slf4j.Slf4j;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.scheduling.annotation.Async;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.stereotype.Component;

import java.io.BufferedInputStream;
import java.io.BufferedReader;
import java.io.DataOutputStream;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.Arrays;

/**
 * Copyright (C), 2022-2023, 姚兵
 * Author: 32210
 * Date: 2023/5/11 14:15
 * FileName: XxlJobHandler
 * Description:
 */

@Component
@Slf4j
public class XxlJobHandler {
    /**
     * XxlJob开发示例（Bean模式）
     *
     * 开发步骤：
     *      1、任务开发：在Spring Bean实例中，开发Job方法；
     *      2、注解配置：为Job方法添加注解 "@XxlJob(value="自定义jobhandler名称", init = "JobHandler初始化方法", destroy = "JobHandler销毁方法")"，注解value值对应的是调度中心新建任务的JobHandler属性的值。
     *      3、执行日志：需要通过 "XxlJobHelper.log" 打印执行日志；
     *      4、任务结果：默认任务结果为 "成功" 状态，不需要主动设置；如有诉求，比如设置任务结果为失败，可以通过 "XxlJobHelper.handleFail/handleSuccess" 自主设置任务结果；
     *
     * @author xuxueli 2019-12-11 21:52:51
     */


        /**
         * 1、简单任务示例（Bean模式）
         */
        @XxlJob("demoJobHandler")
        @Async
        public void demoJobHandler() throws Exception {
            log.info("demoJobHandler:{}","执行了");
            XxlJobHelper.log("XXL-JOB, Hello World.");

            Thread.sleep(4000);

            log.info("demoJobHandler:{}","完成了");
            // default success
        }


        /**
         * 2、分片广播任务
         */
        @XxlJob("shardingJobHandler")
        public void shardingJobHandler() throws Exception {

            // 分片参数
            int shardIndex = XxlJobHelper.getShardIndex();
            int shardTotal = XxlJobHelper.getShardTotal();

            XxlJobHelper.log("分片参数：当前分片序号 = {}, 总分片数 = {}", shardIndex, shardTotal);

            // 业务逻辑
            for (int i = 0; i < shardTotal; i++) {
                if (i == shardIndex) {
                    XxlJobHelper.log("第 {} 片, 命中分片开始处理", i);
                } else {
                    XxlJobHelper.log("第 {} 片, 忽略", i);
                }
            }

        }


        /**
         * 3、命令行任务
         */
        @XxlJob("commandJobHandler")
        public void commandJobHandler() throws Exception {
            String command = XxlJobHelper.getJobParam();
            int exitValue = -1;

            BufferedReader bufferedReader = null;
            try {
                // command process
                ProcessBuilder processBuilder = new ProcessBuilder();
                processBuilder.command(command);
                processBuilder.redirectErrorStream(true);

                Process process = processBuilder.start();
                //Process process = Runtime.getRuntime().exec(command);

                BufferedInputStream bufferedInputStream = new BufferedInputStream(process.getInputStream());
                bufferedReader = new BufferedReader(new InputStreamReader(bufferedInputStream));

                // command log
                String line;
                while ((line = bufferedReader.readLine()) != null) {
                    XxlJobHelper.log(line);
                }

                // command exit
                process.waitFor();
                exitValue = process.exitValue();
            } catch (Exception e) {
                XxlJobHelper.log(e);
            } finally {
                if (bufferedReader != null) {
                    bufferedReader.close();
                }
            }

            if (exitValue == 0) {
                // default success
            } else {
                XxlJobHelper.handleFail("command exit value("+exitValue+") is failed");
            }

        }


        /**
         * 4、跨平台Http任务
         *  参数示例：
         *      "url: http://www.baidu.com\n" +
         *      "method: get\n" +
         *      "data: content\n";
         */
        @XxlJob("httpJobHandler")
        public void httpJobHandler() throws Exception {

            // param parse
            String param = XxlJobHelper.getJobParam();
            if (param==null || param.trim().length()==0) {
                XxlJobHelper.log("param["+ param +"] invalid.");

                XxlJobHelper.handleFail();
                return;
            }

            String[] httpParams = param.split("\n");
            String url = null;
            String method = null;
            String data = null;
            for (String httpParam: httpParams) {
                if (httpParam.startsWith("url:")) {
                    url = httpParam.substring(httpParam.indexOf("url:") + 4).trim();
                }
                if (httpParam.startsWith("method:")) {
                    method = httpParam.substring(httpParam.indexOf("method:") + 7).trim().toUpperCase();
                }
                if (httpParam.startsWith("data:")) {
                    data = httpParam.substring(httpParam.indexOf("data:") + 5).trim();
                }
            }

            // param valid
            if (url==null || url.trim().length()==0) {
                XxlJobHelper.log("url["+ url +"] invalid.");

                XxlJobHelper.handleFail();
                return;
            }
            if (method==null || !Arrays.asList("GET", "POST").contains(method)) {
                XxlJobHelper.log("method["+ method +"] invalid.");

                XxlJobHelper.handleFail();
                return;
            }
            boolean isPostMethod = method.equals("POST");

            // request
            HttpURLConnection connection = null;
            BufferedReader bufferedReader = null;
            try {
                // connection
                URL realUrl = new URL(url);
                connection = (HttpURLConnection) realUrl.openConnection();

                // connection setting
                connection.setRequestMethod(method);
                connection.setDoOutput(isPostMethod);
                connection.setDoInput(true);
                connection.setUseCaches(false);
                connection.setReadTimeout(5 * 1000);
                connection.setConnectTimeout(3 * 1000);
                connection.setRequestProperty("connection", "Keep-Alive");
                connection.setRequestProperty("Content-Type", "application/json;charset=UTF-8");
                connection.setRequestProperty("Accept-Charset", "application/json;charset=UTF-8");

                // do connection
                connection.connect();

                // data
                if (isPostMethod && data!=null && data.trim().length()>0) {
                    DataOutputStream dataOutputStream = new DataOutputStream(connection.getOutputStream());
                    dataOutputStream.write(data.getBytes("UTF-8"));
                    dataOutputStream.flush();
                    dataOutputStream.close();
                }

                // valid StatusCode
                int statusCode = connection.getResponseCode();
                if (statusCode != 200) {
                    throw new RuntimeException("Http Request StatusCode(" + statusCode + ") Invalid.");
                }

                // result
                bufferedReader = new BufferedReader(new InputStreamReader(connection.getInputStream(), "UTF-8"));
                StringBuilder result = new StringBuilder();
                String line;
                while ((line = bufferedReader.readLine()) != null) {
                    result.append(line);
                }
                String responseMsg = result.toString();

                XxlJobHelper.log(responseMsg);

                return;
            } catch (Exception e) {
                XxlJobHelper.log(e);

                XxlJobHelper.handleFail();
                return;
            } finally {
                try {
                    if (bufferedReader != null) {
                        bufferedReader.close();
                    }
                    if (connection != null) {
                        connection.disconnect();
                    }
                } catch (Exception e2) {
                    XxlJobHelper.log(e2);
                }
            }

        }

        /**
         * 5、生命周期任务示例：任务初始化与销毁时，支持自定义相关逻辑；
         */
        @XxlJob(value = "demoJobHandler2", init = "init", destroy = "destroy")
        public void demoJobHandler2() throws Exception {
            XxlJobHelper.log("XXL-JOB, Hello World.");
        }
        public void init(){
            log.info("init");
        }
        public void destroy(){
            log.info("destroy");
        }



}
